import tool.TreeNode;
import tool.TreeNodePrint;
import tool.input;

/**
 * @Description: 合并二叉树
 * 给你两棵二叉树： root1 和 root2 。
 * 想象一下，当你将其中一棵覆盖到另一棵之上时，
 * 两棵树上的一些节点将会重叠（而另一些不会）。
 * 你需要将这两棵树合并成一棵新二叉树。合并的规则是：
 * 如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；
 * 否则，不为 null 的节点将直接作为新二叉树的节点。
 * 返回合并后的二叉树。
 * 注意: 合并过程必须从两个树的根节点开始。
 * @Date:Create in 19:25  2022/8/2
 */
public class mergeTrees0617 {
    public static void main(String[] args) {
        String s1 = "[1,3,2,5,null]";
        String s2 = "[2,1,3,null,4,null,7]";
        TreeNode root1 = input.getTree(s1);
        TreeNode root2 = input.getTree(s2);
        TreeNodePrint.printTree(root1);
        TreeNodePrint.printTree(root2);
        TreeNode root3 = mergeTrees(root1, root2);
        TreeNodePrint.printTree(root3);
    }

    public static TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        return Dfs(root1, root2);
    }

    public static TreeNode Dfs(TreeNode root1, TreeNode root2) {
        if (root1 == null && root2 == null) return null;
        TreeNode root3;
        if (root1 == null && root2 != null) {
            root3 = new TreeNode(root2.val);
            root3.left = Dfs(null, root2.left);
            root3.right = Dfs(null, root2.right);
        } else if (root1 != null && root2 == null) {
            root3 = new TreeNode(root1.val);
            root3.left = Dfs(root1.left, null);
            root3.right = Dfs(root1.right, null);
        } else {
            root3 = new TreeNode(root1.val + root2.val);
            root3.left = Dfs(root1.left, root2.left);
            root3.right = Dfs(root1.right, root2.right);
        }
        return root3;
    }
}
